#!/bin/bash

export CUDA_VISIBLE_DEVICES=1
mkdir -p logs

model_name=timer
token_num=7
token_len=96
seq_len=$[$token_num * $token_len]

# 公共参数
COMMON_ARGS="
  --task_name forecast
  --is_training 1
  --root_path ./dataset/electricity/
  --data_path electricity.csv
  --model_id ECL
  --model $model_name
  --data MultivariateDatasetBenchmark
  --seq_len $seq_len
  --input_token_len $token_len
  --output_token_len $token_len
  --test_seq_len $seq_len
  --e_layers 5
  --d_model 256
  --d_ff 1024
  --batch_size 2
  --learning_rate 0.0005
  --train_epochs 10
  --gpu 0
  --cosine
  --tmax 10
  --use_norm
  --valid_last
"

# 不使用 KAN（MLP）版本
echo ">>> 开始运行 MLP（不使用 KAN）版本"
for pred_len in 96 192 336 720
do
  echo ">>> MLP 预测步长: $pred_len"
  python -u run.py $COMMON_ARGS \
    --test_pred_len $pred_len \
    > logs/mlp_pred_${pred_len}.log 2>&1
done

# 使用 KAN 版本
echo ">>> 开始运行 KAN 版本"
for pred_len in 96 192 336 720
do
  echo ">>> KAN 预测步长: $pred_len"
  python -u run.py $COMMON_ARGS \
    --test_pred_len $pred_len \
    --use_kan \
    > logs/kan_pred_${pred_len}.log 2>&1
done

echo "✅ 所有对比实验完成，日志保存在 logs/ 文件夹中。"
